Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix broadcast add and subtract gradient calculation #2465

Merged
merged 1 commit into from
Jan 19, 2019

Conversation

reminisce
Copy link
Contributor

The gradient calculation of operators add and subtract are not correct when broadcasting is used in the forward pass. Fix the gradient calculation by changing broadcast_to_like to collapse_sum_like in gradient functions of these two ops.

Thanks for contributing to TVM! Please refer to guideline https://docs.tvm.ai/contribute/ for useful information and tips. After the pull request is submitted, please request code reviews from Reviewers.

@tqchen tqchen merged commit eca4f88 into apache:master Jan 19, 2019
@tqchen
Copy link
Member

tqchen commented Jan 19, 2019

Thanks, @reminisce this is now merged. Please request reviewers next time :)

@reminisce
Copy link
Contributor Author

@tqchen Thanks, I will.

Anthony-Mai pushed a commit to Anthony-Mai/tvm that referenced this pull request Jan 21, 2019
Anthony-Mai pushed a commit to Anthony-Mai/tvm that referenced this pull request Jan 28, 2019
Anthony-Mai pushed a commit to Anthony-Mai/tvm that referenced this pull request Jan 28, 2019
wweic pushed a commit to neo-ai/tvm that referenced this pull request Feb 20, 2019
wweic pushed a commit to neo-ai/tvm that referenced this pull request Feb 20, 2019
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants